#
from typing import Dict
from apps.kronos.kronos_engine import KronosEngine

class KronosApp(object):
    def __init__(self):
        self.name = 'apps.kronos.kronos_app.KronosApp'

    @staticmethod
    def startup(params:Dict = {}) -> None:
        print(f'Kronos大型金融预训练模型 v0.0.1')
        if params['task'] == 1:
            # 5分钟K线预测应用
            KronosEngine.predict(params=params)
        if params['task'] == 2:
            KronosEngine.finetune(params=params)